package com.yihusitian.leecode.algorithm.easy;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
 *
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 *
 * 你可以按任意顺序返回答案。
 *
 */
public class TwoSumSolution {


    /**
     * 1.暴力拆解法
     *
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum1(int[] nums, int target) {
        if (Objects.isNull(nums) || nums.length < 2) {
            throw new IllegalArgumentException("参数非法");
        }
        int[] result = new int[2];
        loop: for (int i = 0; i < nums.length; i ++) {
            int current = nums[i];
            for (int j = i + 1; j < nums.length; j ++) {
                if (target - nums[j] == current) {
                    result[0] = i;
                    result[1] = j;
                    break loop;
                }
            }
        }
        return result;
    }

    /**
     * 使用hash表
     *
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum2(int[] nums, int target) {
        if (Objects.isNull(nums) || nums.length < 2) {
            throw new IllegalArgumentException("参数非法");
        }
        int[] result = new int[2];
        Map<Integer, Integer> map = new HashMap<>(nums.length);
        for (int i = 0; i < nums.length; i ++) {
            if (map.size() > 0) {
                int otherValue = target - nums[i];
                int otherIndex = map.getOrDefault(otherValue, -1);
                if (otherIndex != -1) {
                    result[0] = otherIndex;
                    result[1] = i;
                    break;
                }
            }
            map.put(nums[i], i);
        }
        return result;
    }

    /**
     * 使用hash表，一次取两个值，这样可以少遍历几次
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum3(int[] nums, int target) {
        if (Objects.isNull(nums) || nums.length < 2) {
            throw new IllegalArgumentException("参数非法");
        }
        Map<Integer, Integer> map = new HashMap<>(nums.length);
        int left = 0;
        int right = nums.length - 1;

        while (left <= right) {
            int vleft = nums[left];
            int vright = nums[right];
            if (map.containsKey(target - vleft)) {
                return new int[]{map.get(target - vleft), left};
            } else {
                map.put(vleft, left ++);
            }
            if (map.containsKey(target - vright)) {
                return new int[]{map.get(target - vright), right};
            } else {
                map.put(vright, right --);
            }
        }
        return new int[0];
    }

    public static void main(String[] args) {
        int nums[] = new int[]{3,2,4};
        int target = 6;
        new TwoSumSolution().twoSum3(nums, target);
    }

}
